Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[AutoScheduler] Fix task extraction #6965

Merged
merged 4 commits into from
Nov 24, 2020

Conversation

merrymercy
Copy link
Member

@merrymercy merrymercy commented Nov 24, 2020

Follow up of #6956 .
The previous PR does not work for the cases where select_implementation is called during alter op layout in TOPI.

https://github.com/apache/incubator-tvm/blob/b2e418cb109df4cd1f17a2cf2894a1b396a6b838/python/tvm/topi/cuda/conv2d_alter_op.py#L47-L49
We should move the check from lower_call to select_implementation itself.

@merrymercy merrymercy requested a review from comaniac November 24, 2020 04:45
@merrymercy merrymercy merged commit 28b1577 into apache:main Nov 24, 2020
@merrymercy merrymercy deleted the pr-fix-task-extraction branch November 24, 2020 11:22
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 2, 2020
* [AutoScheduler] Fix task extraction

* fix

* fix

* trigger CI
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 4, 2020
* [AutoScheduler] Fix task extraction

* fix

* fix

* trigger CI
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Dec 4, 2020
* [AutoScheduler] Fix task extraction

* fix

* fix

* trigger CI
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants